package com.xiaoming.module2.plugin;

import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.plugin.*;

import java.sql.Connection;
import java.util.Properties;

// 拦截 StatementHandler 的 prepare 方法，指定参数
@Intercepts({
        @Signature(type = StatementHandler.class, method = "prepare",
                args = {Connection.class, Integer.class})
})
public class StatementPlugin implements Interceptor {

    // 拦截方法，目标方法执行时都会执行
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        System.out.println("对StatementHandler进行了增强");
        // 执行后续方法
        return invocation.proceed();
    }

    // 让当前插件生成代理，注册到拦截器链中
    @Override
    public Object plugin(Object target) {
        // target：被拦截的对象，interceptor：自定义实现类
        Object wrap = Plugin.wrap(target, this);
        return wrap;
    }

    // 获取配置文件参数
    @Override
    public void setProperties(Properties properties) {
        System.out.println("获取到的参数：" + properties);
    }
}
